Create prime_factors.py