Wrong comment in def get_country()

There is a incorrect comment in the program assignment and needs to be corrected.

def get_country(city1, country1, city2, embeddings, cosine_similarity=cosine_similarity):

~~

Output:
countries: a dictionary with the most likely country and its similarity score

should be replace as follow

Output:
country : a tuple with the most likely country and its similarity score

It will be corrected. Thank you @beomseok_lee!