Slightly improved docstrings / comments