heatmap working better