Update test.ipynb

Test take 2...