Add stages to DSL