more heat mpi code