Edit functionality added.  Close #1