updated mpi code