Updated the procedure for MPI