prime_factors now works properly