Sort of got classifier to train