added heatmap, works for now