code: public static <T> T augment(WebDriver driver) {
      Augmenter augmenter = new Augmenter();
      augmenter.addDriverAugmentation(IOSCapabilities.CONFIGURABLE, new AddConfigurable());
      augmenter.addDriverAugmentation(IOSCapabilities.ELEMENT_TREE, new AddLogElementTree());
      augmenter.addDriverAugmentation(IOSCapabilities.IOS_SEARCH_CONTEXT, new AddIOSSearchContext());
      augmenter.addDriverAugmentation(IOSCapabilities.IOS_TOUCH_SCREEN, new AddIOSTouchScreen());
      return (T) augmenter.augment(driver);
    }
docs: Add all the IOS specific interfaces ios-driver supports.