data science hw assignment part 2