Added prep_training for balancing training data
Added a classifier training function which is not implemented at all