Cleanup visualizer code.