Minor changes for screenshot