Fixing heatmap margins