Rifaktoroni kodin e mëposhtëm për ta optimizuar atë për shpejtësinë dhe lehtësinë e përdorimit.
def check_prime(num):
    për n në rangun (2, numër):
        nëse num % n == 0:
            kthej False
    kthe e vërtetë
def check_prime(num):
    nëse numër < 2:
        kthej False

    # kontrolloni vetëm numrat tek
    nëse num % 2 == 0:
        kthej False
    
    për n në rangun (3, int(num**0.5)+1, 2):
        nëse num % n == 0:
            kthej False
    ktheje e vërtetë