split out add stage 0 to separate module