Edit the Python program below to add an if condition to determine if an input number is odd or even.
def odd_or_even(number):
    # your code here
def odd_or_even(number):
    # your code here
    if number % 2 == 0:
        return "even"
    else:
        return "odd"