made additional UX improvements