Added docstrings for the module