continue work on multi-output support through shapes