Added functionality to return driver object