Added save and load to plot