Code improvement, PEP8

- PEP8