Add update functionality to procedures