small improvements in ui