Fixing overloading of `best_models`.
Logging model probabilities at the end of each step.
Fixing syntax error from coercing int to string.
Improving logging.