Adding the HMM model class