Clean up visualizer