slight update to homework