some minor fix in UI