- added docs
- code cleanup