Univariate Linear Regression Python