Adding storing/removal of edits.
Refactoring tests.