revised heatmap figure