- change to using a country object