updated script for stress test