Add arguments to main