Data Science week 1 assignment