Changed testTrainHMM to only plot test set