Test.ipynb changes