finished problem 2 draft