Add ability to save graphs