Prime Number Python Code