input: Write a python function to compute the factorial of a number.
  assert factorial(5) == 120
  assert factorial(4) == 24
  assert factorial(1) == 1
output: def factorial(n):
      if n == 0 or n == 1:
          return 1
      else:
          return n * factorial(n - 1)