Create 6. is_prime.py