https://medium.com/@jw1809131/data-science-msc-assignment-homework-help-university-of-edinburgh-edin-uk-ae1e8ad72ed9