Add small documentation
Add tests