Added ability to edit