Assignment for Data Scientist Toolbox class at Coursera