Updating setup_stepwise_benchmark.py to setup SWM and SWA in seperate directories.