Update stress_cpu.py