Updated screenshot.py