Giúp e với cho e xin code python phân tích bài toán tìm số nguyên tố với ah

n = int(input())
check_prime = True
if n<2:
    check_prime = False
else:
    for i in  range(2,int(n**0.5)+1):
        if n%i==0:
            check_prime = False
print(check_prime)