# Small program to increment or decrement

def increment(n):
    return n+1

def decrement(n):
    return n-1

# Only run the code below if this file is directly invoked as
# $ python3 addsubstract.py
if __name__ == "__main__":
    response = input("Do you want to (i)ncrement or (d)ecrement? ")
    number = int(input("What number? "))
    if response == "i":
        result = increment(number)
    else:
        result = decrement(number)
        
    print("Your result is", result)