Add command argument logic