visualization seems to be working
visualizer ipynb is now working