mpi 2d version