homework for The Data Scientist's Toolbox