update visualizer/core.py