updated base_classifier. Still 1 bug