Add update functionality