Assignment for Data Science Class