check_prime_number.py wrapper function change