Small code review with pylint and pep8