Thanks to @pursual the webdriver now has a callback - this fixes #12