Data Scientist's Toolbox: Course Project part 2