optimizing the code in prime_factorization.py