Fix bugs
Visualizer