ui small improvements