More minor improvements to bits of the UI