Determine Prime Number Python Code