add setCountryInput function