add HMM next state prediction