small changes to driver

small changes to driver