small fixes on ui