forgot to add the heatmap class.