updated ipynb