Add missing docstrings

Remove TODOs for tests are tests have been added