small changes to run_driver.py