More minor changes to the UI