working heatmap again