Datascientist assignment 2