Removed diffusion code