https://towardsdatascience.com/the-ultimate-beginners-guide-to-linear-regression-in-python-9f9466c59d73