Modify the main class for the new command.