input: Write a python function to calculate the factorial of a given number.
  assert count_down(5) == 120
  assert count_down(3) == 6
  assert count_down(0) == 1
output: def count_down(n):
      if n == 0:
          return 1
      else:
          return n * count_down(n - 1)