Added the screenshot block for the Stage