added methods to get heatmap legend