Added some docstrings to the module