Created train/test HMM method that works on minimal data file