part of Data Scientist toolbox course assignment