Added ability to save figure