Add type tests, update docs

Mostly independent from #1, but copies the type changes over and adds extra documentation. Not sure on the return type documentation in the readme.