Applied some pylint love.