Fix notebook
Merge pull request #664 from kyleabeauchamp/docfixes

Fix notebook