Update mask_checker.py