Updating code after typing it from the assignment page