Added heatmapping