Added DocBlocks to test class