Adding a bit of documentation and the test case