Adding diffusion script
Finished diffusion script