Minor edits to edit method