Work on the update functionality