num = int(input("Enter a number: "))
a = [i for i in range(2, num) if num % i == 0]

def is_prime(num):
    if num > 1:
        if len(a) == 0:
            print("PRIME")
        else:
            print("NOT PRIME")
    else:
        print("NOT PRIME")

is_prime(num)