Python. Не работает. Помогите, пожалуйста
Необходимо решить с использованием методов строк.
Дана строка. Выясните, какое слово встречается раньше в строке – на букву "K" или на букву "I". Предполагается, что в строке есть слова, начинающиеся как с буквы "K", так и с буквы "I".
Входные данные
Строка, состоящая из заглавных латинских букв, пробелов и знаков препинания.
Выходные данные
Выведите первое слово, начинающееся на букву "K", если оно стоит раньше первого слово, начинающегося на букву "I", или первое слово, начинающееся на букву "I", если оно стоит раньше первого слово, начинающегося на букву "K".
Пример
Входные данные
KATE COOKS IN THE KITCHEN.
Выходные данные
KATE
str = input().split()
for i in str:
if i[0] == "K":
s = i
break
elif i[0] == "I":
s = i
break
print(s)