Add test and change implementation for getScreenShot