minor fixes to UI/UX