Add `n_beams` param?
Add
n_beamsparam?