Алгоритм Евклида

Напишите программу для вычисления НОД двух чисел с помощью Алгоритма Евклида с помощью вычитания.
Входные данные
С клавиатуры вводится два целых числа.
Выходные данные
Вывести одно целое число - НОД.

a = int(input())
b = int(input())
while a != 0 and b != 0:
    if a > b:
        a = a % b
    else:
        b = b % a
print(a + b)