[Fix] Docstrings, typo, and return value