add some UI niceness to the edit screen