fixed formatting issues, added support for screenshot.py

- Fixed formatting issues identified by PEP8 checker, 
- Incorporated comments from @khandelwal 
- Modified class constructors to include a test directory to store images generated by screenshot.py
@OrlandoSoto @khandelwal Those prints are my bad. We can replace them with logger calls. I do not recall why I used prints.