Update for new command