Add stage update option