Adding unit labels to heatmap colorbars