Working on homework 4