https://addpunch.com/linear-regression-python-example/