Prime Number Python Function Example With Variables