Data scientist's toolbox - mini-project 1