Add more tests
Add docs for the new feature
Update the document example
Version bump.