Prime Number Python Function Questions