Hacer un programa que llene un vector de 5 números y calculé cuántos son primos y cuántos no

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

def main():
  a, b = [], []
  for i in range(5):
    num = int(input(“Enter a number: ”))
    if isPrime(num):
      a.append(num)
    else:
      b.append(num)
  print(“Prime:”, a)
  print(“Not Prime:”, b)
    
main()