Single Linear Regression Python Machine Learning