Features/ft 1152 improve heatmap legends