Update screenshot.py