Data scientist's toolkit assignment 1