Update prime_checker.py