Add ability to edit entities