Homework 3 almost done