Using MPI_Cart but not tested yet