Функция сумма цифр числа

#Даны два натуральных числа. В каком из них сумма цифр больше
def sumdigit(number):
    s = 0
    while number > 0:
        s += number % 10
        number //= 10
    return s

a = int(input())
b = int(input())
if sumdigit(a) > sumdigit(b):
    print(a,'больше')
elif sumdigit(b) > sumdigit(a):
    print(b,'больше')
else:
    print('a = b')