UI/UX updates