Coursera course, The Data Scientist's Toolbox, Assignment 1