Add group nodes earlier, but only for multiple samples.
Create group node for IMSR earlier in process