Working on the update method