realizar un programa para saber si un numero es prima

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

# Finds the nth prime number
def nthPrime(n):
  if n == 1:
    return 2
  elif n == 2:
    return 3
  else:
    return nthPrime(n-1