Heatmap functionality added