https://medium.com/@jamesrodrigueswise3/msc-pgdip-pgcert-data-science-assignment-homework-help-newcastle-university-uk-805670acf5f2