Update MPI mesh diagram to be consistent with Firedrake behaviour