UX minor polish