Doing Unit 6 HW. Mixed up in my classes though...