trying to rebalance classes