[MPI] Better output for heat equation problem.