Minor UI fixes and improvements for better user experience.