Add top k viterbi