HW1 - Problem 2 and 3 finished