Refactor stress test