Update driver.go