Add tests
Documentation