A lil more Viterbi train