Cleaning after pep8