Update TEK_driver.py