Stats Linear Regression Python Code