Refactored to use the selenium grid