add ability to specify downstream stage