add visualization ipynb