Move context into the screenshot class