Updating setup_stepwise_benchmark.py to save logs for SWM runs.