Least Squares Regression Python