Visualization improvements