Выведите сумму тех из введённых чисел, у которых из квадраты оканчиваются или на 6,или на 4. Python

# предполагаем, что числа вводятся в одну строку.

array = list(map(int, input().split()))

sum = 0

for item in array:

   if item ** 2 % 10 in [6, 4]:

       sum += item

print(sum)