Make UI a bit nicer