module FFaker::Address

def country(given_code = nil)
  country_index = COUNTRY_CODE.index(given_code)
  if given_code && country_index
    COUNTRY[country_index]
  else
    fetch_sample(COUNTRY)
  end
end