Add beam-search test