Updated homework 4