* better docs, type annotations

* more annotations

* add to README

* make it generic

* fix doc

* complete annotations