I made changes to assignment 1