assignment for Data Science Toolbox course