data_science_assignment_2