2nd part of Assignment for The Data Scientist's Toolbox course