Moved driver to driver.py.