Восстановите программу проверки натурального числа на простоту.
n = int(input())
f = 0
i = 2
z = int(math.sqrt(n))
while i <= z and f == 0:
if n % i == 0:
f = 1
i = i + 1
if f == 1:
print("Число составное")
else:
print("Число простое")