added changes to a driver class