Updated discontinuous plotting so that we can optionally plot a subset of the systems.
This functionality exists for continuous plotting, so good to add it in the
discontinuous case too.