UI/UX modifications