Prime Number Python Function Example