Assignment for Data science