code: def show(filename=None, *args, **kwargs):
          
          if filename is None:
              plt.show(*args, **kwargs)
          else:
              plt.savefig(filename, *args, **kwargs)
docs: Show the current figure.
  
          Parameters
          ----------
          filename : :obj:`str`
              filename to save the image to, for auto-saving