For Assignment 1 in The Data Scientist's Toolbox course