HMM + added confusion plot