PEP8
PEP8
Small refactoring