Update ab_check.py