MPI code for solving 2D the heat diffusion problem