Prime Numbers Python Code