Add stage constructor