Updated Implicit wait to use driver rather than thread.sleep