This is part of "The Data Scientist’s Toolbox" course assignment