Add ability to set equal aspect in heatmap call