Implement web_driver_wait