Pass in driver only in before_screenshot and after_screenshot