Update viterbi.py