Final UI updates and fixes based on user feedback.