Revised Pr/8

Fixes pep8 issues in pr/8