Add some printing for the heatmap.