Составьте программу, определяющую,каких букв, "а" или "о" больше

Предыдущий
вопрос Следующий
вопрос

def count_a_o(string):
count_a = string.count('а')
count_o = string.count('о')
if count_a > count_o:
return 'Букв а больше'
elif count_a < count_o:
return 'Букв о больше'
else:
return 'Количество букв а и о одинаково'
# Пример использования
input_string = input('Введите строку: ')
result = count_a_o(input_string)
print(result)