Traceback viterbi changes
cleanup and minor fixes