fixed formatting issues, added support for screenshot.py