minor changes to ipynb example