Updated argument visitor