Improve visualizations