Second assigment for The Data Scientist's Toolbox course