add the legend of the heatmap