slight UX improvements