Add methods to driver