module FFaker::Address

def country_code(given_country = nil)
  code_index = COUNTRY.index(given_country)
  if given_country && code_index
    COUNTRY_CODE[code_index]
  else
    fetch_sample(COUNTRY_CODE)
  end
end