Add solution and tests for prime_factorization
Be consistent