Add Selenium test code