Updated heatmap visualization