add heatmap stuff