Working on Homework 4