added flag to driver