updated mpiTrain.c