Refactor: Argument setup into initialize