small changes to ipynb