Add new stage for flow