Extract arg setup into overall setup