Allowing a passed in function to run before the screenshot.  Need to finish the test