Save graphs; modify function parameters

Test Plan:
Fix figure creation and paths

Test Plan:
Fix legend

Test Plan: