improve docs (with some [TODO]s)