Update check_args.py