Add main module docstring
Add missing docstrings #8