Simple Python Code For Prime Factors