Adding docs to unit test