Prime number or not

n=int(input('enter number:'))
if(n==1):
    print(n,'not a prime number')
else:
    for i in range(2,n):
        if (n%i==0):
            print(n,'not a prime number')
            break
    else:
        print(n,'prime number')