many updates to demo nb
Closes issue #21, also added some images