layout changes for MPI example