small fixes in UI