https://medium.com/@james53wise/data-science-msc-assignment-homework-help-university-of-surrey-sur-uk-fab805e95ca3