small improvements to the UX