changed heatmap image id to class