working on heatmaps