Python. Задача 2-1

Пользователь вводит два целых числа. Если первое число меньше второго, то вывести сумму чисел, иначе - разность первого и второго.

Решение
a = int(input('Введите число a: '))
b = int(input('Введите число b: '))
if a < b:
    print('Сумма чисел: ', a + b)
else:
    print('Разность чисел: ', a - b)