It would be much nicer if the plot would take up the entire space, similar to what would be the result after plt.tight_layout().