added heatmap caption