Add unit test
Fix documentation format