Помогите решить задачу по питон
Если а>б,то а+б. Если а<б,то а-б

#версия python 3
a = int(input("Введите a: "))
b = int(input("Введите b: "))
if a>b:
    res = a+b
    print(str(res))
elif a<b:
    res = a-b
    print(str(res))
else:
    print("Ошибка")