HW2 for data science