Assignment 1 for The Data Scientist's Toolbox