add some more parallel stage logic