Added convenient method for generating test dataset