Improved visualizations for better interpretability