Add missing documentation
Fix incorrect docstrings