STY: critical pep8 failure
API: generalize to arbitrary keys as a keyword arg to save