Change to homework 4