Clean up main visualization