Update yap_check.py