Added feature to edit