Add arguments to the command function