Cleaning up for screenshot