Update country to country_code