Added behavior to driver class.