Add argument processing