Updating some code to use cssSelector instead of xPath (thanks Galuber)