Some updates to the logging util and run.py