Data science HW4