it's pretty easy to extend `driver` with new methods that could keep this code clean, until the methods are backported to wd. @sebv probably has a comment on how that should be done.