Initialized diffusion.py