prime or not

n = input('What is the nb?')
n= int(n)
i=2
while i <=n and n % i != 0 :
  i = i + 1
if i  == n and n > 1:
  print('Prime!')
else:
  print('not prime')