Update key_driver.rb