Small fixes/cleaning
Added an example notebook