Made edits more persistent - still not perfect