Task 2 of the course Data Science Toolbox