For the Data Science Toolbox course project