This part (b) of the Data Scientist Toolbox assignment