[#115] Clean up

* Add documentation