added test.ipynb notebook