Work on homework 4