working on heatmap