Prime number tester

number = 0
prime = True

number = int( input("What number would you like to test? "))

for i in range (2, number):
    if number % i == 0:
        prime = False

if prime == True:
    print ("Prime")
else:
    print ("Not prime")