Notebook updates

Fixes #129