update to homework 5