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
@khandelwal yes, we had a similar discussion with @marcesher and we decided to break this up into smaller functions, perhaps having one separate function to setup the logger object, etc, so now it's probably a good time for me to do that.