if save_file is None:
        plt.show()
    elif save_file == 'fig':
        return plt.gcf()
    else: