Figure dictionary and save option

- Added dictionary for saving images.
- Added save function to write figures to disk.