Fixes pep8. Tests pass now