For Data Scientist Toolbox homework