Created ScreenShot class. Modified Main class