Almost done with homework 4