Improve edit element dialogs