added plots for class imbalance and beautified existing plots #50

Closed larahabashy closed 3 years ago