Cleaning the code

PEP8 + Pylint