Data Scientist Toolbox exercise