Fix unit testing for test classify (uncommented the train)