Update check environment script