Write a Python function that calculates the factorial of a number using recursion.
==============
def factorial(n):
    if n == 0 or n == 1:
        return 1
    else:
        return n * factorial(n-1)