Adding support of edit