Write docs for public methods

[costash]

Docstrings need to be written to every public method.
Also, the docs for classes should be updated accordingly.