Add logging TODOs
Add docstrings