Added docstrings, little bit more documentation