Add fluent interface to data generator