Check For Prime Number

# check for prime number
num  = int(input("enter a number : "))
if num < 1:
    print("not a prime number")
else:
    for i in range(2, num):
        if num % i == 0:
            print("not a prime number")
            break
    else:
        print("it is prime number")