added some heatmap stuff