added heatmap stuff