Coursera Course - Data Scientist Toolbox assignment