some more tweaks to heatmap