Reature request - plot_loss return its Matplotlib axes for customisation

I was wondering if the Recorder plot_loss function could be slightly modified to use the object oriented Matplotlib functions and return the Matplotlib axes for the plot it renders (and perhaps optionally take a Matplotlib axes parameter to render to).

I noticed when using it that it was a Matplotlib figure - but I was frustrated that unlike the Pandas plot method I wasn’t returning the axes or figure object so that I could customise how it renders (specifically I wanted to make it render to a larger figure size).