Unify plotting behavior

[marinaevers]

All plotting functions should expect an optional parameter that works as a flag to determine if the plot should be shown (plt.show) when calling the function. All functions should return a fig and axs object. The flag should be set to false as a default.