Cleanup docs, add type annotations, improved class ref

Cleanup docs, add type annotations, improved class ref #413