add jsanimation to diffusion notebook