small ux fixes