update to homework 6