Updated HMM training