Added countryCode function