Part of the project for Data Science Toolbox course