Add pep8 compliance