added control for screenshot