Fixed stress.py