Given a natural number N. Determine if it is prime. A natural number N is called prime if it has only two divisors: one and the number N itself. As an answer, print "Yes" if the number is prime, "No" otherwise.

def prime(n):
  if n == 1:
    return False
  elif n == 2:
    return True
  for i in range(2, n):
    if n % i == 0:
      return False
  return True
  
prime(2)