improve heatmaps