Задача на Python. Прошу помочь. Это очень срочно..

Код решения задачи на Python:

n = int(input())  # количество строк для проверки
word = input()  # проверочное слово

count = 0  # счетчик подходящих строк

for _ in range(n):
    line = input()
    if "забыл" in line or word in line:
        count += 1

if count <= n / 2:
    print("НЕ ТАК И МНОГО")
else:
    print("ВСЕ РАВНО НИЧЕГО СТРАШНОГО")

print(count)

Пример работы программы:

Входные данные:
5
шапка
Бильбо забыл шапку
Бильбо забыл трость
Бильбо оставил посуду невымытой
Бильбо забыл деньги
Бильбо забыл носовой платок

Выходные данные:
ВСЕ РАВНО НИЧЕГО СТРАШНОГО
4

Входные данные:
3
кольцо
Бильбо забыл шапку
Бильбо забыл кольцо
Бильбо забыл трость

Выходные данные:
НЕ ТАК И МНОГО
2