Improved code quality

Using pylint