Move edit behavior to begin_edit fn