You may want to move this to the helpers module, providing a driver argument for reusability