Add test case
Add docs