Fix wrapper tests
Add documentation