Update diffusion.js