upated for new command arg