Updated HMM trainer to work with the new ann api