Add heatmap analysis