Added working 2D diffusion code and modified output method.