Число простое или сложное?

Вводится натуральное число. Программа должна определять, является ли оно простым или сложным.

import math

n = int(input())
if n < 2:
    print("A number must be more 1")
    quit()
elif n == 2:
    print("It's prime number")
    quit()
i = 2
limit = int(math.sqrt(n))
while i <= limit:
    if n % i == 0:
        print("It's composite number")
        quit()
    i += 1
print("It's prime number")