Continue working on the update functionality