for the project of "The Data Scientist's Toolbox" course