Data science homework 1