updates from class
finish assignment 1