Data science homework