Add missing doc string and remove pylint warnings