slight changes to how the figure is saved