def factorial(n):
    if n == 0 or n == 1:
        return 1
    else:
        return n * factorial(n - 1)

# Example usage
num = 2
print(f"Factorial of {num} is {factorial(num)}")