UI/UX enhancements and error handling