Updated take_screenshot code since in version 2.42 the class
RemoteWebDriver implements TakesScreenshot