Assignment 1 for Data Scientist Toolbox