Data science homework 1 solutions