Little pylint warning fixes