improve edit buttons