Fix loading of HMMs
Only train when necessary