passed the first question in HW4... still need to clean things up