really simple python program to generate prime factors