Add test for new command