Single Linear Regression Python Code