Add docstrings

See issue #2.
Sucks