Minor documentation tweaks

- Added some docstrings
- Added the (more detailed) descriptions from README as docstrings for methods