Simple Linear Regression Code Python