small fixes in ui