Cleaned code

Cleaned code with pylint.