Assignment 2 for Data Science