Add method to generate toy dataset