Prime Number Python Function Exercises