Add DataWrapper class