https://dosenit.com/python/linear-regression-python