few improvements in docstrings and stuff :relieved: