Add support for plotting in the notebook

Closes gh-4