Update function_is_prime.py