Refactored tests, removed duplication.
Viterbi algorithm works with returning the tagging sequence as well. Needs to be refactored though.