Fix for pep8 compliance