Fix tests
Add docs