Add graph selection
Add graph saving