adding legend to the heatmap