Assignment 2 of data scientist's toolbox