Use a smaller example dataset