add docstrings to module/class
add docstrings and setup.py