rework testing (adding a ipynb)