Added edit and cleaned up cli
Added a test for edits