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