DOC : update docstrings and comments