More minor edits

More ui stuff