added docstrings, fixed pip install bug