update some docstrings, more documentation