Edit command fix for #6
Merge pull request #16 from mtahmed/edit-comment

Edit command fix for #6