Add stage number to new/edit stage form