Refactor for better useability. Docstrings
Integrate docstrings