wip: add docs, cleanup