https://medium.com/@jamesrodrigueswise/machine-learning-in-science-msc-assignment-homework-help-university-of-nottingham-uk-329015ff105c