Check if number is prime

a = int(input("enter number to check:      "))

def is_prime(n):
    for i in range(2,n):
        if n%i == 0:
            return False
    return True
result = is_prime(a)
if result == False:
    print("not prime")
else:
    print("prime")