Task 2 of Data Scientist Toolbox Project