Add update functionality.