These are the changes to assignment 1