fixed bug in diffusion stencil code