added some more docs
added some docstrings