additional functionality for edit mode