add correct classification outputs
fix viterbi and add more classification output