Exercise for 'The Data Scientist's Toolbox'