Updates to homework 3