For Data Scientists' Toolbox assignment 1