For "The Data Scientist’s Toolbox" course project