* add docstrings
* pylint?
