Diffusion Sample Code