Homework 4 almost complete