Ошибка в коду питхон!!!!

 a = 0 
b = 0 
n = int(input('n: ')) 
for j in range(n): 
    c = int(input('c: ')) 
    b += c 
    if c <= 40: 
        a += 1 
print(b / n) 
if 2 <= a: 
    print('YES') 
else: 
    print('NO')