Enhanced heatmap visualization