i believe i achieved the result of an exercise with a smaller code

i fulfilled the following excerise question diffrently than the given solution and i wanted to know if this was correct or if i missed something.

the question:
an you remember how to check if a key exists in a dictionary?
Using the capitals dictionary below write some code to ask the user to input
a country, then check to see if that country is in the dictionary and if it is
print the capital. If not tell the user it's not there.

solution i provided:
country = input("Input a country \n>> ")
country = country.capitalize()
if country in capitals.keys():
     print(capitals[country])
elif country.isalpha() == False:
     print("error please type a string")
else:
     print("Country not found in dictionary")

answer provided by the instructor:
#capitals = {'France':'Paris','Spain':'Madrid','United Kingdom':'London',
#            'India':'New Delhi','United States':'Washington DC','Italy':'Rome',
#            'Denmark':'Copenhagen','Germany':'Berlin','Greece':'Athens',
#            'Bulgaria':'Sofia','Ireland':'Dublin','Mexico':'Mexico City'
#            }
#
#user_input = input('Which country would you like to check?:> ')
#
#user_input = user_input.lower()
#
#
#
#while ('united kingdom' not in user_input and not user_input.isalpha()):
#    if user_input == 'united states':
#        break
#    print('You must input a string')
#    user_input = input('Which country would you like to check?:> ')
#
#user_input = user_input.title()
##print(user_input)
#if user_input in capitals:
#    print(f'The capital of {user_input} is {capitals[user_input]}')
#else:
#    print('No data available')