https://www.sohamkamani.com/data-science/linear-regression-python/