Add some docstrings
Add usage documentation