finding prime factors in python