square root regression python