finish problem 2

Did what I could for HW 7