TestWrapper class added