Add init_args() function