Specify explicit grid_shape for examples created via dist.

This PR adds an explicit `grid_shape` argument for the plot examples, which matches the original default shape assignment. This ensures that if the default arrangement changes, as PR #274 now does, that the plot examples will stay the same.