Use country_field instead of a custom implementation.
Updated documentation.