Prime Number Python Formula