homework for Data Science Toolbox