Right now the ensemble stuff works for getting the underlying dynamics (i.e., the u) but doesn't provide the z from the solutions. Might be nice to have a way to get that in the ensemble simulations as well. For example, a setting to concatenate them/etc.