Modified to use one instance of webdriver