Create linear_regression.py