Allowing for different distances in axes_grid