Add new code for Selenium Test