[MOD] Fix pep8