seems to be correctly training the HMM now...