Follow up on #83 to add support for notebooks. Tricky part will be clearing the UI state when this happens.
