Added next_n_tokens method