Update hs_driver.py