Minor UI updates and bug fixes.