Made code pep8 conforming