Update check_env.py