Return fig instead of axes in plot()

[mtazzari]

So that the usage is similar to that of the corner package.
Inside plot the axes can be accessed with fig.gca() and from the user perspective, it is simpler to do:

fig = uv.plot()
fig = uv.plot(fig=fig)
fig.savefig("uvplot.png")