Add class imbalance check and new print format to robust_logistic_regression.py