ui small fixes