Add tests
Add test docs