Cleanup notebook/fix function name

Fixes #149