https://towardsdatascience.com/supervised-machine-learning-using-linear-regression-part1-ed59d0136be8