wip and docstrings