Numerical solution to the heat equation in 2 dimensions