Add comments to test class