UX updates and bug fixes.