add visualizer code