Update driver_script.py