Added docstring, fixed pylint suggested style