Amend linearregression.py