Prettier, but still no heatmap