Update linear_regression.py