Removed the screenshot code