Update prime_factors.py

Added in conditions for n=1 and n=2. Needs testing.