Heatmap tweaks