Added initial version of driver.py