Fix for pep8