Added some docstrings and changed exports