several minor fix per review

* added docstrings
* move import
* fix typo in dependency