Added a driver test for the screenshot method