added heatmaps