[FIX] Fix pep8 errors