axis crop on the heatmap function