Prime Number Python Function Return