Minor changes in notebook
Create a python module from diffusion notebook
Add figure comparing different diffusion models