Add --edit option