Added docstrings a bit