stress: Improve the stress test scripts