heatmap works
fix stuff