add: docstrings and fix TU