Added methods to get the webdriver object for certain elements