Updates to the heatmaps