homework for The Data scientist's Toolbox