changes to plot saving