removing some debug code from driver.py