Linear Regression Example Code Python