small fixes to improve UI