Simple Linear Regression Python Code