manca un

codice:
return country;
come ultima riga della funzione