add heatmap stuff (w missing files)