add test for screenshot