Implement support for edit mode