tidy up and notes added to homework