pylint improvments