fixed change to arg parsing