Update driver.py

Updated documentation.