Linear Regression Coding In Python