PEP8 & PEP257 cleanups

Added pep257 comments and made pep8 fixes