Add updates for stage