Added heatmap visualization