updates to homework 6