small improvements on the ui