Atomic diffusion updates code