Ускорить код

[AndreyNill]

Здравствуйте, имею код, выдает ответ правильно, но не проходит по времени в тестирующей системе. Как мне его изменить, чтобы код стал работать быстрее?

def isPrime(n):
    if n in [2,3]:
        return("YES")
    else:
        if n%2==0:
            return("NO")
        else:
            for i in range(3,n//2+1,2):
                if n%i==0:
                    return("NO")
            return("YES")
n = int(input())
print(isPrime(n))

 

[stud_55]

Можно попробовать после return("NO") поставить break чтобы цикл прекращался сразу после нахождения первого делителя, а не шел до конца.