add more tests
Add some docs
Add some docs