Updated viterbi