Added tests for prime_factorization