Determine Prime Number Python Function