slight ux fixes