pylint improvements