✨ Return fig and axes from plotting functions

[s-weigand]

Is your feature request related to a problem? Please describe.

Users might want to further tweak plots after creating them, so the objects should be returned.

Describe the solution you'd like

Just add a return statement at the end of plot functions returning the axes and figure (only needed if newly created)