Proper notebook of Diffusion Modelling