just need to fix axes on heatmap