small improvements on the UI