Class Balancing = DONE
TODO Sprint 2