Can you create a program in Python that will return the nth Fibonacci number?
n = 8
def fib(n):
    if n == 0 or n == 1:
        return 1
    else:
        return fib(n-1) + fib(n-2)

n = 8
print(fib(n))