Improvements to visualizations