Added simple diffusion
Cleanup