Update stress.py