ste function and linear diffusion code