Add documentation for test class