Add more implicit wait for javascript stuff in Selenium test