Fix HMM example