Now producing square heatmaps