working on heatmap stuff