heatmap classes fix 3