Improved styling according to PEP
Moved screenshots into separate branch