improving the country selection ui