Linear Regression Single Variable Python