add plt.savefig