Modifying viterbi code to take in more arguments, since I'll need to iteratively run and train.  It might be slightly slower in total, but that's OK, since my computer is slow anyway.