Add missing docstrings, improve others