Data Scientist Toolbox first assignment