homework assignement for Data Scientist's Toolbox