Fix tests
Improve docs