added docstrings
minor changes
version 0.1.1