Improve hacks/stress.py a bit.