Diffusion code/burgers1 played with trying to simplify